package com.mokairui.linear;

import java.util.Iterator;

/**
 * @Description 栈
 * @Author Mokairui
 * @Since 2022/3/26
 */
public class Stack<T> implements Iterable<T> {
    // 记录首结点
    private Node head;
    // 栈中元素的个数
    private int N;
    
    public Stack() {
        this.head = new Node(null, null);
        this.N = 0;
    }
    
    // 判断当前栈中元素个数是否为0
    public boolean isEmpty(){
        return N == 0;
    }
    
    // 把t元素压入栈
    public void push(T t){
        // 找到首节点指向的第一个节点
        Node oldFirst = head.next;
        // 创建新节点
        Node newNode = new Node(t, null);
        // 让首节点指向新的节点
        head.next = newNode;
        // 让新节点指向原来的第一个节点
        newNode.next = oldFirst;
        // 元素个数+1
        N++;
    }
    
    // 弹出栈顶元素
    public T pop(){
        // 找到首节点指向的第一个节点
        Node oldFirst = head.next;
        if (oldFirst == null) {
            return null;
        }
        // 让首节点指向原来第一个节点的下一个节点,也就是取消调用当前的第一个节点
        head.next = oldFirst.next;
        // 元素个数-1
        N--;
        return oldFirst.item;
    }

    // 获取栈中元素的个数
    public int size(){
        return N;
    }
    
    @Override
    public Iterator<T> iterator() {
        return new SIterator();
    }
    
    private class SIterator implements Iterator<T>{
        private Node n = head;
        @Override
        public boolean hasNext() {
            return n.next != null;
        }
        @Override
        public T next() {
            Node node = n.next;
            n = n.next;
            return node.item;
        }
    }
    
    private class Node{
        public T item;
        public Node next;
        public Node(T item, Node next) {
            this.item = item;
            this.next = next;
        }
    }
}
